package com.example.lcpractice.lc;

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;
import java.util.Stack;

/**
 * [739]每日温度.java
 * <p>
 * 给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。
 * <p>
 * 示例 1:
 * <p>
 * 输入: temperatures = [73,74,75,71,69,72,76,73]
 * 输出: [1,1,4,2,1,1,0,0]
 * 示例 2:
 * <p>
 * 输入: temperatures = [30,40,50,60]
 * 输出: [1,1,1,0]
 * 示例 3:
 * <p>
 * 输入: temperatures = [30,60,90]
 * 输出: [1,1,0]
 * 提示：
 * <p>
 * 1 <= temperatures.length <= 105
 * 30 <= temperatures[i] <= 100
 * Related Topics
 * 栈
 * 数组
 * 单调栈
 */
public class Lc739 {
    public static void main(String[] args) {
        Stack<Integer> deque = new Stack<>();
        deque.add(1);
        deque.add(2);
        deque.add(4);
        deque.add(55);
        deque.add(8);
        System.out.println(deque);
        deque.pop();
        System.out.println(deque.peek());

        System.out.println(Arrays.toString(dailyTemperatures2(new int[]{2,1,3,2})));
    }

    /**
     * 暴力遍历
     * @param temperatures
     * @return
     */
    public static int[] dailyTemperatures(int[] temperatures) {
        int[] waits = new int[temperatures.length];
        for (int i = 0; i < temperatures.length; i++) {
            for (int j = 1 + i; j < temperatures.length; j++) {
                if (temperatures[j] > temperatures[i]) {
                    waits[i] = j - i;
                    break;
                } else if (j == temperatures.length - 1) {
                    waits[j] = 0;
                }
            }
        }
        return waits;
    }

    public static int[] dailyTemperatures2(int[] T) {
        int n = T.length;
        //用于保存结果
        int[] ans = new int[n];
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < n; i++) {
            // 条件：堆栈不为空，当前数比栈首序数对应数字大
            // 隐含的入栈条件是 入栈代表它不比已有的前面的大
            // 一旦比前面数字大，则挨个回溯到所有能大的，防止解法一的重复遍历
            while (!stack.empty() && T[i] > T[stack.peek()]) {
                ans[stack.peek()] = i - stack.peek();
                stack.pop();
            }
            // 每一个序号都是要进栈的，在上面的出栈操作中，每找到一个比栈尾的都赋值给结果，并且出栈了
            stack.push(i);
        }
        return ans;
    }

    public static int[] dailyTemperatures3(int[] T) {
        Deque<Integer> deque = new ArrayDeque<>();
        deque.push(0);
        for (int i = 1; i < T.length; i++) {
            if (deque.isEmpty()) {
                deque.push(i);
                // 双队列的第一入栈
            } else if (T[deque.peek()] >= T[i]) {
                deque.push(i);
            } else {
                while (true) {
                    if (deque.isEmpty() || T[deque.peek()] >= T[i]) {
                        deque.push(i);
                        break;
                    }
                    T[deque.peek()] = i - deque.pop();
                }
            }
        }
        while (!deque.isEmpty()) {
            T[deque.pop()] = 0;
        }
        return T;
    }
}

